import numpy as np


if __name__ == "__main__":
    seed_path = "/home/ubuntu/Ascend/dataset/tensorflow/seed/tf.nn.softmax_cross_entropy_with_logits" \
                "/20221129_032847/tf.nn.softmax_cross_entropy_with_logits_seeds.npz"
    seed_dict = np.load(seed_path)
    x = seed_dict["labels_340"]
    y = seed_dict["logits_340"]
    print(x)
    print(x.shape)
    print(x.dtype)
    print(y)
    print(y.shape)
    print(y.dtype)
    inputs = {
        "labels_1": x, "logits_1": y
    }
    np.savez("SoftmaxCrossEntropyWithLogits_inputs", **inputs)
